
from models.ultrasam import UltraSAM

def main(image_path):
    model = UltraSAM()
    mask_path = model.infer(image_path)
    print("推理完成，输出：", mask_path)

if __name__ == "__main__":
    import sys
    import os
    image_path = sys.argv[1] if len(sys.argv) > 1 else "test.png"
    main(image_path)
